Saving data to a file in Python